package com.atguigu.tingshu.user.service.impl;

import com.alibaba.fastjson.JSON;
import com.atguigu.tingshu.common.constant.SystemConstant;
import com.atguigu.tingshu.common.rabbit.constant.MqConst;
import com.atguigu.tingshu.common.rabbit.service.RabbitService;
import com.atguigu.tingshu.common.util.MongoUtil;
import com.atguigu.tingshu.model.user.UserListenProcess;
import com.atguigu.tingshu.user.service.UserListenProcessService;
import com.atguigu.tingshu.vo.album.TrackStatMqVo;
import com.atguigu.tingshu.vo.user.UserListenProcessVo;
import org.bson.types.ObjectId;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Sort;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;

import java.math.BigDecimal;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;
import java.util.UUID;
import java.util.concurrent.TimeUnit;

@Service
@SuppressWarnings({"all"})
public class UserListenProcessServiceImpl implements UserListenProcessService {

    @Autowired
    private MongoTemplate mongoTemplate;

    @Autowired
    private RedisTemplate redisTemplate;

    @Autowired
    private RabbitService rabbitService;

    /**
     * 获取用户播放声音进度 如果mongoDB中有声音进度 直接返回进度，否则返回0
     *
     * @param trackId
     * @param userId
     * @return
     */
    @Override
    public BigDecimal getTrackBreakSecond(Long trackId, Long userId) {
        // 根据用户id，声音id设置查询条件
        Query query = Query.query(Criteria.where("trackId").is(trackId).and("userId").is(userId));
        UserListenProcess userListenProcess = mongoTemplate.findOne(query, UserListenProcess.class
                , MongoUtil.getCollectionName(MongoUtil.MongoCollectionEnum.USER_LISTEN_PROCESS, userId));
        if (null != userListenProcess) {
            // 如果查询结果不为空，则返回用户听歌进度
            return userListenProcess.getBreakSecond();
        }
        // 如果查询结果为空，则返回0
        return new BigDecimal(0);
    }

    /**
     * 更新用户播放声音进度  点击获取声音进度的时候，如果mongoDB中有声音进度，则更新进度，否则创建新的UserListenProcess对象，并设置属性
     *
     * @param userListenProcessVo
     * @param userId
     */
    @Override
    public void updateListenProcess(UserListenProcessVo userListenProcessVo, Long userId) {
        // 根据用户id，声音id设置查询条件
        Query query = Query.query(Criteria.where("trackId").is(userListenProcessVo.getTrackId())
                .and("userId").is(userId));
        String collectionName = MongoUtil.getCollectionName(MongoUtil.MongoCollectionEnum.USER_LISTEN_PROCESS, userId);
        UserListenProcess userListenProcess = mongoTemplate.findOne(query, UserListenProcess.class, collectionName);
        if (null != userListenProcess) {
            // 如果查询结果不为空，则更新breakSecond和updateTime
            userListenProcess.setBreakSecond(userListenProcessVo.getBreakSecond());
            userListenProcess.setUpdateTime(new Date());
//            mongoTemplate.save(userListenProcess, collectionName);
        } else {
            // 如果查询结果为空，则创建新的UserListenProcess对象，并设置属性
            userListenProcess = new UserListenProcess();
            BeanUtils.copyProperties(userListenProcessVo, userListenProcess);
            userListenProcess.setId(ObjectId.get().toString());
            userListenProcess.setUserId(userId);
            userListenProcess.setIsShow(1);
            userListenProcess.setCreateTime(new Date());
            userListenProcess.setUpdateTime(new Date());
        }
        // 保存UserListenProcess对象到MongoDB
        mongoTemplate.save(userListenProcess, collectionName);
        // 根据用户id，声音id设置 redis缓存key，记录专辑与声音播放量，同一个用户同一个声音每天只记录一次播放量
        String key = "user:" + userId + "track:" + userListenProcessVo.getTrackId();
        Boolean ifAbsent = redisTemplate.opsForValue().setIfAbsent(key, 1, 24, TimeUnit.HOURS);
        if (ifAbsent) {
            // 如果缓存中不存在该key，则发送消息到mq
            TrackStatMqVo trackStatMqVo = new TrackStatMqVo();
            // 设置业务编号 防止消息重复消费
            trackStatMqVo.setBusinessNo(UUID.randomUUID().toString().replaceAll("-", ""));
            trackStatMqVo.setAlbumId(userListenProcessVo.getAlbumId());
            trackStatMqVo.setTrackId(userListenProcessVo.getTrackId());
            trackStatMqVo.setStatType(SystemConstant.TRACK_STAT_PLAY);
            trackStatMqVo.setCount(1);
            //	异步发送消息，更新专辑-声音播放量
            rabbitService.sendMessage(MqConst.EXCHANGE_TRACK, MqConst.ROUTING_TRACK_STAT_UPDATE, JSON.toJSONString(trackStatMqVo));
        }
    }

    /**
     * 根据用户id，查询mongoDB中最近播放的声音
     * @param userId
     * @return
     */
    @Override
    public Map<String, Object> getRecentTrack(Long userId) {
        Query query = Query.query(Criteria.where("userId").is(userId)).with(Sort.by(Sort.Direction.DESC, "updateTime"));
        // 根据用户id，查询mongoDB中最近播放的声音
        String collectionName = MongoUtil.getCollectionName(MongoUtil.MongoCollectionEnum.USER_LISTEN_PROCESS, userId);
        UserListenProcess userListenProcess = mongoTemplate.findOne(query, UserListenProcess.class, collectionName);
        HashMap<String, Object> map = new HashMap<>();
        if (null != userListenProcess) {
            map.put("trackId", userListenProcess.getTrackId());
            map.put("albumId", userListenProcess.getAlbumId());
            return map;
        }
        return map;
    }
}
